# -*- coding:utf-8 -*-
from com.mat.rpa.utils.globalConstants import GlobalConstants
from com.mat.rpa.views.workWindow.leftPanel.directiveTree import treeItem
from . import AIGlobalConstants
from  .characterRecognition import characterRecognitionSubDirectiveTree
from .NLPOperation import NLPOperationSubDirectiveTree
class AISubDirectiveTree(treeItem.TreeItemWithID):
    def __init__(self, parent):
        super(AISubDirectiveTree, self).__init__(parent)
        self.createAISubTreeNode()

    def createAISubTreeNode(self):
        self.nodeType = GlobalConstants.treeBranchType
        self.directiveType = "AIOperation"
        self.setText(0, "人工智能AI")
        self.createTreeNode(GlobalConstants.treeLeafType,
                            self, AIGlobalConstants.AIGlobalConstants.creatingAIEngineDirective, "创建AI引擎")
        #挂载子树
        characterRecognitionSubDirectiveTree.CharacterRecognitionSubDirectiveTree(self)
        NLPOperationSubDirectiveTree.NLPOperationSubDirectiveTree(self)